import type { BasicTableProps, TableActionType, FetchParams, BasicColumn } from '../types/table';
import type { PaginationProps } from '../types/pagination';
import type { DynamicProps } from '/#/utils';
import type { FormActionType } from '/@/components/Form';
import type { WatchStopHandle } from 'vue';
import { getDynamicProps } from '/@/utils';
import { ref, onUnmounted, unref, watch, toRaw } from 'vue';
import { isProdMode } from '/@/utils/env';
import { error } from '/@/utils/log';

type Props = Partial<DynamicProps<BasicTableProps>>;

type UseTableMethod = TableActionType & {
  getForm: () => FormActionType;
};

export function useTable(tableProps?: Props): [
  (instance: TableActionType, formInstance: UseTableMethod) => void,
  TableActionType & {
    getForm: () => FormActionType;
  },
] {
  // 创建一个响应式引用，用于存储表格实例
  const tableRef = ref<Nullable<TableActionType>>(null);
  // 创建一个响应式引用，用于标记表格是否已加载
  const loadedRef = ref<Nullable<boolean>>(false);
  // 创建一个响应式引用，用于存储表单实例
  const formRef = ref<Nullable<UseTableMethod>>(null);

  // 定义一个停止观察者的句柄
  let stopWatch: WatchStopHandle;

  // 注册表格实例和表单实例的函数
  function register(instance: TableActionType, formInstance: UseTableMethod) {
    // 如果是生产模式，则在组件卸载时清空表格实例和加载状态
    isProdMode() &&
      onUnmounted(() => {
        tableRef.value = null;
        loadedRef.value = null;
      });

    // 如果表格已加载且为生产模式且实例与已注册的实例相同，则直接返回
    if (unref(loadedRef) && isProdMode() && instance === unref(tableRef)) return;

    // 更新表格实例和表单实例
    tableRef.value = instance;
    formRef.value = formInstance;
    // 如果传入了表格属性，则更新表格属性
    tableProps && instance.setProps(getDynamicProps(tableProps));
    // 标记表格已加载
    loadedRef.value = true;

    // 停止之前的观察者
    stopWatch?.();

    // 创建一个新的观察者，用于监听表格属性的变化
    stopWatch = watch(
      // 观察表格属性
      () => tableProps,
      // 当表格属性变化时，更新表格属性
      () => {
        tableProps && instance.setProps(getDynamicProps(tableProps));
      },
      // 观察者选项
      {
        immediate: true, // 立即执行回调
        deep: true,       // 深度观察
      },
    );
  }

  // 获取表格实例的函数
  function getTableInstance(): TableActionType {
    const table = unref(tableRef);
    // 如果没有获取到表格实例，则抛出错误
    if (!table) {
      error('The table instance has not been obtained yet, please make sure the table is presented when performing the table operation!');
    }
    // 返回表格实例
    return table as TableActionType;
  }

  // 定义表格操作方法
  const methods: TableActionType & {
    getForm: () => FormActionType;
  } = {
    // 重新加载表格数据
    reload: async (opt?: FetchParams) => {
      return await getTableInstance().reload(opt);
    },
    // 设置表格属性
    setProps: (props: Partial<BasicTableProps>) => {
      getTableInstance().setProps(props);
    },
    // 重新计算表格高度
    redoHeight: () => {
      getTableInstance().redoHeight();
    },
    // 设置选中的行
    setSelectedRows: (rows: Recordable[]) => {
      return toRaw(getTableInstance().setSelectedRows(rows));
    },
    // 设置表格加载状态
    setLoading: (loading: boolean) => {
      getTableInstance().setLoading(loading);
    },
    // 获取表格数据源
    getDataSource: () => {
      return getTableInstance().getDataSource();
    },
    // 获取原始表格数据源
    getRawDataSource: () => {
      return getTableInstance().getRawDataSource();
    },
    // 获取表格请求参数
    getFetchParams: () => {
      return getTableInstance().getFetchParams();
    },
    // 获取表格列
    getColumns: ({ ignoreIndex = false }: { ignoreIndex?: boolean } = {}) => {
      const columns = getTableInstance().getColumns({ ignoreIndex }) || [];
      return toRaw(columns);
    },
    // 设置表格列
    setColumns: (columns: BasicColumn[]) => {
      getTableInstance().setColumns(columns);
    },
    // 设置表格数据
    setTableData: (values: any[]) => {
      return getTableInstance().setTableData(values);
    },
    // 设置分页信息
    setPagination: (info: Partial<PaginationProps>) => {
      return getTableInstance().setPagination(info);
    },
    // 根据键删除选中的行
    deleteSelectRowByKey: (key: string) => {
      getTableInstance().deleteSelectRowByKey(key);
    },
    // 获取选中的行键
    getSelectRowKeys: () => {
      return toRaw(getTableInstance().getSelectRowKeys());
    },
    // 获取选中的行
    getSelectRows: () => {
      return toRaw(getTableInstance().getSelectRows());
    },
    // 清除选中的行键
    clearSelectedRowKeys: () => {
      getTableInstance().clearSelectedRowKeys();
    },
    // 设置选中的行键
    setSelectedRowKeys: (keys: string[] | number[]) => {
      getTableInstance().setSelectedRowKeys(keys);
    },
    // 获取分页引用
    getPaginationRef: () => {
      return getTableInstance().getPaginationRef();
    },
    // 获取表格尺寸
    getSize: () => {
      return toRaw(getTableInstance().getSize());
    },
    // 更新表格数据
    updateTableData: (index: number, key: string, value: any) => {
      return getTableInstance().updateTableData(index, key, value);
    },
    // 删除表格数据记录
    deleteTableDataRecord: (rowKey: string | number | string[] | number[]) => {
      return getTableInstance().deleteTableDataRecord(rowKey);
    },
    // 插入表格数据记录
    insertTableDataRecord: (record: Recordable | Recordable[], index?: number) => {
      return getTableInstance().insertTableDataRecord(record, index);
    },
    // 更新表格数据记录
    updateTableDataRecord: (rowKey: string | number, record: Recordable) => {
      return getTableInstance().updateTableDataRecord(rowKey, record);
    },
    // 查找表格数据记录
    findTableDataRecord: (rowKey: string | number) => {
      return getTableInstance().findTableDataRecord(rowKey);
    },
    // 获取行选择信息
    getRowSelection: () => {
      return toRaw(getTableInstance().getRowSelection());
    },
    // 获取缓存的列
    getCacheColumns: () => {
      return toRaw(getTableInstance().getCacheColumns());
    },
    // 获取表单实例
    getForm: () => {
      return unref(formRef) as unknown as FormActionType;
    },
    // 设置是否显示分页
    setShowPagination: async (show: boolean) => {
      getTableInstance().setShowPagination(show);
    },
    // 获取是否显示分页
    getShowPagination: () => {
      return toRaw(getTableInstance().getShowPagination());
    },
    // 展开所有行
    expandAll: () => {
      getTableInstance().expandAll();
    },
    // 展开指定行
    expandRows: (keys: string[]) => {
      getTableInstance().expandRows(keys);
    },
    // 折叠所有行
    collapseAll: () => {
      getTableInstance().collapseAll();
    },
    // 获取是否展开
    getIsExpanded: () => {
      return getTableInstance().getIsExpanded();
    },
    // 滚动到指定位置
    scrollTo: (pos: string) => {
      getTableInstance().scrollTo(pos);
    },
  };

  // 返回注册函数和方法对象
  return [register, methods];
}
